import allure
from airtest.core.api import exists
from airtest.core.cv import Template

from common.airtest.airtest_base import AirtestBase
from common.log.log import log
from config.conf import BASE_DIR


class MyShare:
    """个人中心-我的分享页面"""

    @allure.step('个人中心-我的分享页面：点击提现说明')
    def click_tips(self):
        """
        个人中心-我的分享页面：点击提现说明
        """
        AirtestBase().click_5(BASE_DIR + r"/element_picture/我的/我的分享/提现规则按钮.png")

    @allure.step('个人中心-我的分享页面：点击立即提现，并提现')
    def immediate_withdrawal(self):
        """
        个人中心-我的分享页面：点击立即提现，并提现
        return：成功或失败（只要提现ok即可）
        """
        AirtestBase().click_5(BASE_DIR + r"/element_picture/我的/我的分享/立即提现.png")
        AirtestBase().click_5(BASE_DIR + r"/element_picture/我的/我的分享/申请提现/全部提现.png")
        AirtestBase().click_5(BASE_DIR + r"/element_picture/我的/我的分享/申请提现/确认提现.png")

        """
        零元提现
        """
        element_1 = BASE_DIR + r"/element_picture/我的/我的分享/申请提现/不再提现日期范围内.png"
        element_2 = BASE_DIR + r"/element_picture/我的/我的分享/申请提现/提现成功提示.png"
        if exists(Template(element_1)):
            log.info('不在提现日期！')
            return True
        elif exists(Template(element_2)):
            log.info('提现成功！')
            return True
        else:
            return False

    @allure.step('个人中心-我的分享页面：点击推广名片')
    def click_promotion_business_card(self):
        """
        个人中心-我的分享页面：点击推广名片
        """
        AirtestBase().click_5(BASE_DIR + r"/element_picture/我的/我的分享/推广名片.png")

    @allure.step('个人中心-我的分享页面：打开推广人统计，点击“查看我的上级”')
    def click_promoter_statistics(self):
        """
        个人中心-我的分享页面：打开推广人统计，点击“查看我的上级”
        """
        AirtestBase().click_5(BASE_DIR + r"/element_picture/我的/我的分享/推广人统计.png")
        AirtestBase().click_5(BASE_DIR + r"/element_picture/我的/我的分享/推广人统计/查看我的上级.png")

    @allure.step('个人中心-我的分享页面：点击佣金记录”')
    def click_commission_record(self):
        """
        个人中心-我的分享页面：点击佣金记录
        """
        AirtestBase().click_5(BASE_DIR + r"/element_picture/我的/我的分享/佣金记录.png")

    @allure.step('个人中心-我的分享页面：点击提现记录”')
    def click_withdrawal_record(self):
        """
        个人中心-我的分享页面：点击提现记录
        """
        AirtestBase().click_5(BASE_DIR + r"/element_picture/我的/我的分享/提现记录.png")

    @allure.step('个人中心-我的分享页面：点击推广素材库”')
    def click_material_library(self):
        """
        个人中心-我的分享页面：点击推广素材库
        """
        AirtestBase().click_5(BASE_DIR + r"/element_picture/我的/我的分享/推广素材库.png")
